package com.zy.aspectj;

import com.alibaba.fastjson2.JSONObject;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.zy.common.annotation.Dict;
import com.zy.common.core.domain.AjaxResult;
import com.zy.common.core.page.TableDataInfo;
import com.zy.common.utils.DictUtils;
import com.zy.common.utils.ObjConvertUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;

@Aspect
@Component
@Slf4j
public class DictAspect {

    //翻译后拼接的内容
    private static String DICT_TEXT_SUFFIX = "_mc";

    // 定义切点Pointcut 拦截所有对服务器的请求
    @Pointcut("execution( * com.zy.web.controller..*.*(..))")
    public void excudeService() {
    }

    /**
     * 这是触发 excudeService 的时候会执行的
     *
     * @param pjp
     * @return
     * @throws Throwable
     */
    @Around("excudeService()")
    public Object doAround(ProceedingJoinPoint pjp) throws Throwable {
        //这是定义开始事件
        long time1 = System.currentTimeMillis();
        //这是方法并获取返回结果
        Object result = pjp.proceed();
        //这是获取到 结束时间
        long time2 = System.currentTimeMillis();
        log.debug("获取JSON数据 耗时：" + (time2 - time1) + "ms");
        //解析开始时间
        long start = System.currentTimeMillis();
        //开始解析（翻译字段内部的值凡是打了 @Dict 这玩意的都会被翻译）
        this.parseDictText(result);
        //解析结束时间
        long end = System.currentTimeMillis();
        log.debug("解析注入JSON数据  耗时" + (end - start) + "ms");
        return result;
    }

    /**
     * 本方法针对返回对象为Result 的IPage的分页列表数据进行动态字典注入
     * 字典注入实现 通过对实体类添加注解@dict 来标识需要的字典内容,字典分为单字典code即可 ，table字典 code table text配合使用与原来jeecg的用法相同
     * 示例为SysUser   字段为sex 添加了注解@Dict(dicCode = "sex") 会在字典服务立马查出来对应的text 然后在请求list的时候将这个字典text，已字段名称加_dictText形式返回到前端
     * 例输入当前返回值的就会多出一个sex_dictText字段
     * {
     * sex:1,
     * sex_dictText:"男"
     * }
     * 前端直接取值sext_dictText在table里面无需再进行前端的字典转换了
     * customRender:function (text) {
     * if(text==1){
     * return "男";
     * }else if(text==2){
     * return "女";
     * }else{
     * return text;
     * }
     * }
     * 目前vue是这么进行字典渲染到table上的多了就很麻烦了 这个直接在服务端渲染完成前端可以直接用
     *
     * @param result
     */
    private void parseDictText(Object result) {
        if(com.zy.common.utils.StringUtils.isNull(result)){
            return;
        }
        List<JSONObject> items = new ArrayList<>();
        List<Object> list = new ArrayList<>();
        TableDataInfo pageUtils = null;
        AjaxResult dataUtils = null;
        if (result instanceof TableDataInfo) {
            pageUtils = (TableDataInfo) result;
            list = (List<Object>)pageUtils.getRows();
        }else {
            dataUtils = (AjaxResult)result;
            if(dataUtils.get("data")!=null){
                if(dataUtils.get("data") instanceof List){
                    list = (List)dataUtils.get("data");
                }else {
                    list.add(dataUtils.get("data"));
                }
            }else{
                return;
            }
        }
        //循环查找出来的数据
        for (Object record : list) {
            ObjectMapper mapper = new ObjectMapper();
            String json = "{}";
            try {
                //解决@JsonFormat注解解析不了的问题详见SysAnnouncement类的@JsonFormat
                json = mapper.writeValueAsString(record);
            } catch (JsonProcessingException e) {
                log.error("json解析失败" + e.getMessage(), e);
            }
            JSONObject item = JSONObject.parseObject(json);

            //update-begin--Author:scott -- Date:20190603 ----for：解决继承实体字段无法翻译问题------
            //for (Field field : record.getClass().getDeclaredFields()) {
            for (Field field : ObjConvertUtils.getAllFields(record)) {
                //update-end--Author:scott  -- Date:20190603 ----for：解决继承实体字段无法翻译问题------
                if (field.getAnnotation(Dict.class) != null) {
                    String code = field.getAnnotation(Dict.class).dictDataSource();
                    String text = field.getAnnotation(Dict.class).dictText();
                    //获取当前带翻译的值
                    String key = String.valueOf(item.get(field.getName()));
                    //翻译字典值对应的txt
                    String textValue = translateDictValue(code, key);
                    //  CommonConstant.DICT_TEXT_SUFFIX的值为，是默认值：
                    // public static final String DICT_TEXT_SUFFIX = "_dictText";
                    //如果给了文本名
                    if (!StringUtils.isEmpty(text)) {
                        item.put(text, textValue);
                    } else {
                        //走默认策略
                        item.put(field.getName() + DICT_TEXT_SUFFIX, textValue);
                    }

                }
                //date类型默认转换string格式化日期
                /*if (field.getType().getName().equals("java.util.Date") && field.getAnnotation(JsonFormat.class) == null && item.get(field.getName()) != null) {
                    SimpleDateFormat aDate = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
                    item.put(field.getName(), aDate.format(new Date((Long) item.get(field.getName()))));
                }*/
            }
            items.add(item);
        }
        if (result instanceof TableDataInfo) {
            pageUtils.setRows(items);
        }else {
            if(dataUtils.get("data") instanceof List){
                dataUtils.put("data",items);
            }else {
                dataUtils.put("data",items.get(0));
            }

        }
    }


    /**
     * 翻译字典文本
     *
     * @param code
     * @param key
     * @return
     */
    private String translateDictValue(String code, String key) {
        //如果key为空直接返回就好了
        if (ObjConvertUtils.isEmpty(key)) {
            return null;
        }
        StringBuffer textValue = new StringBuffer();
        //分割 key 值
        //System.out.println(code+":::::"+key);
        String[] keys = key.split(",");
        //循环 keys 中的所有值
        for (String k : keys) {
            String tmpValue = null;
            if (k.trim().length() == 0) {
                continue; //跳过循环
            }
            tmpValue = DictUtils.getDictLabel(code,k.trim());

            if (tmpValue != null) {
                if (!"".equals(textValue.toString())) {
                    textValue.append(",");
                }
                textValue.append(tmpValue);
            }
        }
        //返回翻译的值
        return textValue.toString();
    }

}
